#!/usr/bin/python
# -*- coding: utf-8 -*-
# @name             : 文件压缩
# @author           : OneYoung
# @create           : 2024/02/04
# @version          : v1.0
# @desc             : 将文件夹压缩为ZIP文件

import os
import zipfile
import shutil


def zip_file(folder_path, zip_path):
    """
    将指定文件保存为zip文件
    :param folder_path: 需要压缩的文件夹路径
    :param zip_path: 文件保存路径
    :return:
    """
    with zipfile.ZipFile(zip_path, 'w') as zip_obj:
        # 循环遍历文件夹中的所有文件和子文件夹
        for root, dirs, files in os.walk(folder_path):
            for file in files:
                # 将每个文件添加到zip文件中
                print(root, dirs, file)
                zip_obj.write(os.path.join(root, file))


def zip_folder(folder_path, zip_path):
    """
    将指定文件夹保存为zip文件
    :param folder_path: 需要压缩的文件夹路径
    :param zip_path: 文件保存路径
    :return:
    """
    shutil.make_archive(zip_path, 'zip', folder_path)
